#!/bin/bash 
#SBATCH -J task_param
#SBATCH -p gpu-normal 
#SBATCH -N 1 
#SBATCH --gres=gpu:1
#SBATCH -n 1
#SBATCH -o task_param.out 
#SBATCH -e task_param.err


module load anaconda3/2019.10 cuda/10.1

cd ..
pwd

python train.py --gpu_id=0 --output_dir=output_param --is_load=True --iteration=5000 --impute_iter=2000 --log_interval=100 --gru_type=grui --num_layer=2  --seq_len=12 --hidden_dim=48  --w_lambda2=0.05 --w_eta=100 --w_zeta=100

python train.py --gpu_id=0 --output_dir=output_param --is_load=True --iteration=5000 --impute_iter=2000 --log_interval=100 --gru_type=grui --num_layer=2  --seq_len=6 --hidden_dim=48  --w_lambda2=0.05 --w_eta=100 --w_zeta=100

python train.py --gpu_id=0 --output_dir=output_param --is_load=True --iteration=5000 --impute_iter=2000 --log_interval=100 --gru_type=grui --num_layer=2  --seq_len=6 --hidden_dim=96  --w_lambda2=0.05 --w_eta=100 --w_zeta=100
